scikit-learn==1.0.1
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
torch==1.10.0
torchvision==0.11.1
pyyaml==6.0
matplotlib==3.4.3
seaborn==0.11.2
tqdm==4.62.3
opencv-python==4.5.4.58
